<?php
/**
 * @title 658. 找到 K 个最接近的元素 - 二分查找
 * @author start2004
 */

// ini_set("memory_limit", "128M");

// class Solution {
//
//     function find() {
//
//     }
// }


class Solution {

    /**
     * @param Integer[] $arr
     * @param Integer $k
     * @param Integer $x
     * @return Integer[]
     */
    function findClosestElements($arr, $k, $x) {
        /**
         * @since 2022-08-25 长度
         */
        $len = count($arr);

        /**
         * @since 2022-08-25 结果集
         */
        $result = [];

        /**
         * @since 2022-08-25 比最小值更小
         */
        if($x <= $arr[0]){
            for ($i=0; $i<$k; $i++){
                $result[] = $arr[$i];
            }
        } elseif($x >= $arr[$len-1]) {
            /**
             * @since 2022-08-25 比最大值更大
             */
            for($i=$len-$k; $i<$len; $i++){
                $result[] = $arr[$i];
            }
        } else {
            /**
             * @since 2022-08-25 二分查找，<=x 的最大数
             */
            $s = 0;
            $e = $len-1;
            while( ($e-$s) > 1){
                $m = ($s+$e)>>1;
                $v = $arr[$m];
                // echo "[{$s}, {$e}], m={$m}, v={$v}, __";

                if($v == $x){
                    $s = $m;
                    break;
                } elseif($v > $x) {
                    $e = $m;
                } else {
                    $s = $m;
                }
                // echo "[{$s}, {$e}]", PHP_EOL;
            }
            // echo "[{$s}, {$e}]__", PHP_EOL;die();

            /**
             * @since 2022-08-25 双指针
             */
            $p1 = $s;
            $p2 = $s+1;
            for($i=0; $i<$k; $i++){
                /**
                 * @since 2022-08-25 左边p1到头，全部从右边p2取值
                 */
                if($p1 < 0){
                    while($i < $k){
                        $result[] = $arr[$p2];
                        $p2++;
                        $i++;
                    }
                    break;
                } elseif($p2 >= $len) {
                    /**
                     * @since 2022-08-25 右边p2到头，全部从左边p1取值
                     */
                    while($i < $k){
                        array_unshift($result, $arr[$p1]);
                        $p1--;
                        $i++;
                    }
                    break;
                } else {
                    /**
                     * @since 2022-08-25 左右比较
                     */
                    $abs1 = abs($arr[$p1]-$x);
                    $abs2 = abs($arr[$p2]-$x);

                    /**
                     * @since 2022-08-25 取更小值
                     */
                    if($abs1 <= $abs2){
                        array_unshift($result, $arr[$p1]);
                        $p1--;
                    } else {
                        $result[] = $arr[$p2];
                        $p2++;
                    }
                }
            }
        }

        /**
         * @return
         */
        return $result;
    }
}



/**
 * @url http://127.0.0.1/leetcode/202208/2022.08.25.php
 */
$datas = [
    // [[1,2,4,5], 2, 3], // 2,4
    // [[1,2,3,4,5], 4, 3], // 1,2,3,4
    // [[1,2,3,4,5], 4, -1], // 1,2,3,4
    // [[1,2,3,4,5], 4, 10], // 2,3,4,5
    [[0,1,1,1,2,3,6,7,8,9], 9, 4],
];

include_once dirname(__DIR__) . DIRECTORY_SEPARATOR ."xhprof.php";
$xhprof = new Xhprof();
foreach ($datas as $data){
    if(count($data) <= 20){
        var_dump($data);
    } else {
        var_dump(count($data));
    }

    $obj = new Solution();
    $result = $obj->findClosestElements(...$data);
    // $result = $obj->($xhprof->tree($data));
    // $result = $obj->($xhprof->listNode($data));
    // $result = $obj->find(...$data);
    var_dump($result);
    // if(count($result)<=20){
    //     var_dump($result);
    // } else {
    //     var_dump(count($result));
    // }
    echo str_repeat(PHP_EOL, 3);
    echo str_repeat("<br>", 3);
}

// foreach ($datas as $data){
//     $obj = new $data[0][0](...$data[1][0]);
//
//     for ($i=1; $i<count($data[0]); $i++){
//         $func = $data[0][$i];
//         $param = $data[1][$i];
//         $result = $obj->$func(...$param);
//
//         echo $data[0][$i] ."(\"". implode(",", $data[1][$i]) ."\") ";
//             if($result === true){
//                 echo "True";
//             } elseif($result === false) {
//                 echo "False";
//             } elseif($result === null) {
//                 echo "Null";
//             } elseif(is_array($result)) {
//                 var_dump($result);
//             } else {
//                 echo $result;
//             }
//         echo PHP_EOL;
//     }
//
//     echo str_repeat(PHP_EOL, 3);
// }
$xhprof->end();
